package com.xj.algorithm.leetcode;

/**
 * 反转字符串 II
 */
public class L541_分组反转字符串 {

    /**
     * 给定一个字符串和一个整数 k，你需要对从字符串开头算起的每个 2k 个字符的前k个字符进行反转。如果剩余少于 k 个字符，则将剩余的所有全部反转。如果有小于 2k 但大于或等于 k 个字符，则反转前 k 个字符，并将剩余的字符保持原样。
     *
     * 示例:
     *
     * 输入: s = "abcdefg", k = 2
     * 输出: "bacdfeg"
     * 要求:
     *
     * 该字符串只包含小写的英文字母。
     * 给定字符串的长度和 k 在[1, 10000]范围内。
     */
    public static void main(String[] args) {
        System.out.println(reverseStr("abcdefg", 2));
        System.out.println(reverseStrOffice("abcdefg", 2));
    }

    //自己的解法：拆分字符转，逐个处理。使用stringBuffer的反转函数
    public static String reverseStr(String s, int k) {
        String res = "";

        int length = s.length();
        for (int i = 0; i < length; i = i + 2 * k) {
            int right = i + 2 * k >= length ? length : i + 2 * k;
            String tmp = s.substring(i, right);
            if (tmp.length() >=k) {
                res = res + new StringBuffer(tmp.substring(0, k)).reverse().toString()
                        + tmp.substring(k);
            } else if (tmp.length() < k) {
                res = res + new StringBuffer(tmp).reverse().toString();
            }
        }
        return res;
    }

    //官方解法：暴力解法,伪2重循环
    public static String reverseStrOffice(String s, int k) {
        char[] a = s.toCharArray();
        for (int start = 0; start < a.length; start += 2 * k) {
            int i = start, j = Math.min(start + k - 1, a.length - 1);
            while (i < j) {
                char tmp = a[i];
                a[i++] = a[j];
                a[j--] = tmp;
            }
        }
        return new String(a);
    }

}
